cd ../../../
exp_id="newmynet_multi_plus"
dataset='hico'
dataset_path='data/hico_20160224_det'
addr="127.0.0.1"
port=1119
num_gpu=4
backbone='resnet50'
pretrained='params/Detr/detr-r50-pre-2stage-q64.pth'
output_dir=logs/$dataset/$exp_id/stage1/

export OMP_NUM_THREADS=1
python main.py \
--world_size $num_gpu \
--master_addr $addr \
--master_port $port \
--exp_id $exp_id \
--output_dir $output_dir \
--dataset_file $dataset \
--hoi_path $dataset_path \
--backbone $backbone \
--pretrained $pretrained \
--num_obj_classes 80 \
--num_verb_classes 117 \
--dec_layers_hopd 6 \
--dec_layers_interaction 6 \
--batch_size 4 \
--lr 1e-4 \
--epochs 20 \
--lr_drop 10 \
--num_queries 100 \
--use_nms_filter \
--k_one2many 4